from rest_framework import viewsets
from .models import Dorm, DormAssignment
from .serializers import DormSerializer, DormAssignmentSerializer

class DormViewSet(viewsets.ModelViewSet):
    queryset = Dorm.objects.all()
    serializer_class = DormSerializer

class DormAssignmentViewSet(viewsets.ModelViewSet):
    queryset = DormAssignment.objects.all()
    serializer_class = DormAssignmentSerializer